// 实现异步日志缓冲区
#ifndef __M_BUF_H__
#define __M_BUF_H__
#include "util.hpp"
#include <vector>
#include <cassert>

namespace mylog
{
#define DEFAULT_BUFFER_SIZE (1 * 1024 * 1024)   // 缓冲区的默认大小
#define THRESHOLD_BUFFER_SIZE (8 * 1024 * 1024) // 缓冲区大小的阈值，小于阈值翻倍扩容，达到阈值线性扩容
#define INCREMENT_BUFFER_SIZE (1 * 1024 * 1024) // 线性扩容每次扩容的大小
    class Buffer
    {
    public:
        Buffer() : _buffer(DEFAULT_BUFFER_SIZE), _reader_idx(0), _writer_idx(0) {}

        // 向缓冲区写入数据
        void push(const char *data, size_t len)
        {
            // 缓冲区剩余空间不足时考虑扩容
            ensureEnoughSize(len);
            // 将数据拷贝进缓冲区
            std::copy(data, data + len, &_buffer[_writer_idx]);
            // 将当前写入位置向后偏移
            moveWriter(len);
        }

        // 返回可读数据起始位置
        const char *begin()
        {
            return &_buffer[_reader_idx];
        }

        // 返回可读数据长度
        size_t readableSize()
        {
            // 因为当前实现的是双缓冲区，处理完就交换，不存在空间的循环利用（_writer_idx < _reader_idx）
            return _writer_idx - _reader_idx;
        }

        // 返回可写数据长度
        size_t writeableSize()
        {
            return _buffer.size() - _writer_idx;
        }

        // 对读指针进行向后便偏移操作
        void moveReader(size_t len)
        {
            assert(len <= readableSize());
            _reader_idx += len;
        }

        // 读写位置重置，初始化缓冲区
        void reset()
        {
            _reader_idx = _writer_idx = 0;
        }

        // 对缓冲区实现交换操作
        void swap(Buffer &buffer)
        {
            _buffer.swap(buffer._buffer);
            std::swap(_reader_idx, buffer._reader_idx);
            std::swap(_writer_idx, buffer._writer_idx);
        }

        // 判断缓冲区是否为空
        bool empty()
        {
            return _reader_idx == _writer_idx;
        }

    private:
        // 对缓冲区进行扩容
        void ensureEnoughSize(size_t len)
        {
            if (len <= writeableSize()) // 可写空间大小充足，不需要扩容
            {
                return;
            }
            size_t new_size = 0;
            while (len > new_size)//防止len很大的情况下，扩容后len还大于new_size
            {
                if (_buffer.size() < THRESHOLD_BUFFER_SIZE) // 小于阈值翻倍扩容
                {
                    new_size = _buffer.size() * 2;
                }
                else // 达到阈值线性扩容
                {
                    new_size = _buffer.size() + INCREMENT_BUFFER_SIZE;
                }
            }
            _buffer.resize(new_size);
        }

        // 对写指针进行向后便偏移操作
        void moveWriter(size_t len)
        {
            assert(len <= writeableSize());
            _writer_idx += len;
        }

    private:
        std::vector<char> _buffer;
        size_t _reader_idx; // 当前可读数据的下标
        size_t _writer_idx; // 当前可写数据的下标
    };
}
#endif